﻿/*
 * Description: 显示分辨率设置修改
 */

#include <Windows.h>
#include <string>
#include <tchar.h>

const int ARGC_COUNT = 2;
int main(int argc, char* argv[])
{
    if (argc == ARGC_COUNT && strcmp(argv[1], "restore") == 0) {
        int select =  MessageBoxW(NULL,
        L"检测到系统驱动存在异常，修复重启后可正常使用。\n是否立即修复并重启？\n",
        L"虚拟机管理助手",
        MB_ICONQUESTION | MB_YESNO);
        if (select == IDYES) {
            return 1;
        } else {
            return 0;
        }
     } else {
        int select =  MessageBoxW(NULL,
        L"检测到系统需要更新驱动及相关服务，更新后自动重启可正常使用。\n是否立即更新？\n",
        L"虚拟机管理助手",
        MB_ICONQUESTION | MB_YESNO);
        if (select == IDYES) {
            return 1;
        } else {
            return 0;
        }
    }
}